// Copyright (c) 2024 Huawei Technologies Co., Ltd.
// openFuyao is licensed under Mulan PSL v2.
// You can use this software according to the terms and conditions of the Mulan PSL v2.
// You may obtain a copy of Mulan PSL v2 at:
//          http://license.coscl.org.cn/MulanPSL2
// THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
// EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
// MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
// See the Mulan PSL v2 for more details.

package lm

import (
	corev1 "k8s.io/api/core/v1"
)

const (
	hostArchLabelKey = "host-arch"
)

func newHostArchLabeler(node *corev1.Node) Labeler {
	labels := Labels{}

	arch, ok := node.Labels["kubernetes.io/arch"]
	if !ok {
		return Labels{hostArchLabelKey: "unknown"}
	}

	if arch == "amd64" {
		labels[hostArchLabelKey] = "huawei-x86"
	} else {
		labels[hostArchLabelKey] = "huawei-arm"
	}

	return labels
}
